package com.writing.dto;

import lombok.Data;
import java.util.List;

/**
 * AI文本生成请求
 */
@Data
public class AIGenerateRequest {

    /**
     * 提示词内容
     */
    private String prompt;

    /**
     * 系统提示词
     */
    private String systemPrompt;

    /**
     * 对话历史（可选）
     */
    private List<Message> messages;

    /**
     * 模型名称（例如：gpt-3.5-turbo, gpt-4）
     */
    private String model;

    /**
     * 最大token数
     */
    private Integer maxTokens;

    /**
     * 温度参数（0-2）
     */
    private Double temperature;

    /**
     * Top P参数
     */
    private Double topP;

    /**
     * 频率惩罚
     */
    private Double frequencyPenalty;

    /**
     * 存在惩罚
     */
    private Double presencePenalty;

    /**
     * 生成类型（generation/chat等）
     */
    private String type;

    /**
     * 对话消息
     */
    @Data
    public static class Message {
        private String role;
        private String content;
    }
}

